package com.chengcang.api.modules.sys.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.chengcang.api.modules.sys.dao.SysProductDao;
import com.chengcang.api.modules.sys.dao.SysProductTypeDao;
import com.chengcang.api.modules.sys.dto.SysProductTypeDTO;
import com.chengcang.api.modules.sys.entity.SysProductTypeEntity;
import com.chengcang.api.modules.sys.service.SysProductTypeService;
import com.chengcang.api.modules.sys.vo.SysProductTreeVo;
import com.chengcang.common.service.impl.CrudServiceImpl;
import com.chengcang.common.utils.ConvertUtils;
import com.chengcang.common.utils.TreeUtils;
import com.google.common.collect.Lists;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * 
 *
 * @author Zhanli xy321231@163.com
 * @since 1.0.0 2023-01-14
 */
@Service
public class SysProductTypeServiceImpl extends CrudServiceImpl<SysProductTypeDao, SysProductTypeEntity, SysProductTypeDTO> implements SysProductTypeService {

    @Autowired
    private SysProductTypeDao sysProductTypeDao;

    @Autowired
    private SysProductDao sysProductDao;

    @Override
    public QueryWrapper<SysProductTypeEntity> getWrapper(Map<String, Object> params){
        String id = (String)params.get("id");

        QueryWrapper<SysProductTypeEntity> wrapper = new QueryWrapper<>();
        wrapper.eq(StringUtils.isNotBlank(id), "id", id);

        return wrapper;
    }

    public List<Long> getSecondIdsById(Long id){
        try{
            return sysProductTypeDao.selectList( new LambdaQueryWrapper<SysProductTypeEntity>()
                    .eq(SysProductTypeEntity::getPid,id)).stream().map(SysProductTypeEntity::getId).collect(Collectors.toList());
        }catch (Exception e){
            log.error(e.getMessage(), e);
            throw new RuntimeException("查询二级分类异常!");
        }
    }

    @Override
    public List<SysProductTypeDTO> getTree() {
        List<SysProductTypeEntity> allType = sysProductTypeDao.selectList(new LambdaQueryWrapper<SysProductTypeEntity>());
        List<SysProductTypeDTO> sysProductTypeDTOS = ConvertUtils.sourceToTarget(allType, SysProductTypeDTO.class);
        sysProductTypeDTOS.forEach(item -> {
            if (item.getChildren().size() == 0) {
                item.setChildren(Lists.newArrayList(new SysProductTypeDTO()));
            }
        });
        return TreeUtils.build(sysProductTypeDTOS);
    }

    @Override
    public List<SysProductTreeVo> getAllTree() {
        //type
        List<SysProductTreeVo> productTrees = sysProductTypeDao.getAll();
        //product
        List<SysProductTreeVo> products = sysProductDao.getAll();

        //2.23 产品说暂时由后台过滤数据
        //second types
        List<SysProductTreeVo> seconds = productTrees.stream().filter(item -> {
            return item.getPid() != null;
        }).collect(Collectors.toList());

        //use second ids
        List<Long> useSecondIds = products.stream().map(SysProductTreeVo::getPid).distinct().collect(Collectors.toList());

        //use fist ids
        List<Long> useFirstIds = seconds.stream().filter(item -> {
            return useSecondIds.contains(item.getId());
        }).map(SysProductTreeVo::getPid).distinct().collect(Collectors.toList());

        List<SysProductTreeVo> allType = productTrees.stream().filter(item -> {
            return useFirstIds.contains(item.getId()) || useSecondIds.contains(item.getId());
        }).collect(Collectors.toList());

        //总
        ArrayList<SysProductTreeVo> all = new ArrayList<>();
        all.addAll(allType);
        all.addAll(products);
        return TreeUtils.build(all);

//        for(int i=0; i < result.size(); ){
//            //一级产品`
//            SysProductTreeVo sysProductTreeVo = result.get(i);
//            //二级 列表
//            List seconds = sysProductTreeVo.getChildren();
//            if(seconds.size() == 0){
//                result.remove(i);
//            }else {
//                for(int j=0; j < seconds.size(); ){
//                    //二级 产品
//                    SysProductTreeVo second = (SysProductTreeVo)seconds.get(j);
//                    if(second.getChildren().size() == 0){
//                        seconds.remove(j);
//                    }
//                    ++j;
//                }
//            }
//            ++i;
//        }
//
//        for(int i=0; i < result.size(); ){
//            //一级产品
//            SysProductTreeVo sysProductTreeVo = result.get(i);
//            //二级 列表
//            List seconds = sysProductTreeVo.getChildren();
//            if(seconds.size() == 0){
//                result.remove(i);
//            }
//            ++i;
//        }
    }

}